/*
 *Copyright 2021-2023 NERCIS
 *
 *Licensed under the Apache License, Version 2.0 (the "License");
 *you may not use this file except in compliance with the License.
 *You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 *Unless required by applicable law or agreed to in writing, software
 *distributed under the License is distributed on an "AS IS" BASIS,
 *WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *See the License for the specific language governing permissions and
 *limitations under the License.
 */

package cn.ac.nercis.pes.service.project.imported;

import cn.ac.nercis.pes.model.dto.common.Result;
import cn.ac.nercis.pes.model.dto.project.ImportConflictDTO;
import cn.ac.nercis.pes.model.standard.prepare.PasswordServerSDTO;
import cn.ac.nercis.pes.repository.dal.evaluation.PasswordServerRepository;
import cn.ac.nercis.pes.repository.model.evaluation.PasswordServerDO;
import cn.ac.nercis.pes.service.utils.CheckFieldUtils;
import lombok.NonNull;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.time.LocalDateTime;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.stream.Collectors;

/**
 * 密码服务信息导入服务
 *
 * @author zpy
 * @version 1.0.0
 */
@Slf4j
@Service
public class PasswordServerImportService {
    private final PasswordServerRepository passwordServerRepository;

    @Autowired
    public PasswordServerImportService(PasswordServerRepository passwordServerRepository) {
        this.passwordServerRepository = passwordServerRepository;
    }

    public Result<List<ImportConflictDTO>> checkPasswordServer(@NonNull String systemId, List<PasswordServerSDTO> passwordServers) {
        List<PasswordServerDO> sourceData = this.passwordServerRepository.findAllBySystemId(systemId);
        if (CollectionUtils.isEmpty(sourceData)) {
            return Result.success();
        }
        List<ImportConflictDTO> conflictData = passwordServers.stream()
                .flatMap(ec -> this.checkPasswordServer(ec, sourceData).stream())
                .filter(Objects::nonNull)
                .collect(Collectors.toList());
        return Result.success(conflictData);
    }

    public Result<String> importPasswordServer(@NonNull String systemId, List<PasswordServerSDTO> passwordServers, Boolean cover) {
        List<PasswordServerDO> sourceData = this.passwordServerRepository.findAllBySystemId(systemId);
        Result<String> result = Result.success();
        for (PasswordServerSDTO item : passwordServers) {
            result = this.importPasswordServer(systemId, item, sourceData, cover);
            if (!result.isSuccess()) {
                break;
            }
        }
        return result;
    }

    private List<ImportConflictDTO> checkPasswordServer(@NonNull PasswordServerSDTO importData, List<PasswordServerDO> sourceData) {
        PasswordServerDO entity = sourceData.stream()
                .filter(st -> StringUtils.equals(st.getName(), importData.getName()))
                .findFirst().orElse(null);
        if (Objects.isNull(entity)) {
            return Collections.emptyList();
        }
        return CheckFieldUtils.checkField(entity, importData);
    }

    private Result<String> importPasswordServer(String systemId, @NonNull PasswordServerSDTO importData, List<PasswordServerDO> sourceData, Boolean cover) {
        PasswordServerDO entity = sourceData.stream()
                .filter(st -> StringUtils.equals(st.getName(), importData.getName()))
                .findFirst().orElse(null);
        if (Objects.isNull(entity)) {
            return createPasswordServer(systemId, importData);
        } else {
            return modifyPasswordServer(entity, importData, cover);
        }
    }

    private Result<String> createPasswordServer(String systemId, PasswordServerSDTO importData) {
        PasswordServerDO entity = PasswordServerDO.builder()
                .systemId(systemId)
                .name(importData.getName())
                .vendor(importData.getVendor())
                .build();
        entity.setCreateTime(LocalDateTime.now());
        return Optional.of(entity)
                .map(this.passwordServerRepository::save)
                .map(PasswordServerDO::getId)
                .map(Result::success)
                .orElse(Result.failed(String.format("导入%s密码服务信息，创建失败", importData.getName())));
    }

    private Result<String> modifyPasswordServer(PasswordServerDO sourceData, PasswordServerSDTO importData, Boolean cover) {
        sourceData.setVendor(CheckFieldUtils.applyContent(sourceData.getVendor(), importData.getVendor(), cover));
        sourceData.setModifyTime(LocalDateTime.now());
        return Optional.of(sourceData)
                .map(this.passwordServerRepository::save)
                .map(PasswordServerDO::getId)
                .map(Result::success)
                .orElse(Result.failed(String.format("导入%s密码服务信息，修改失败", importData.getName())));
    }
}
